# import unittest
# from case import test_demo01, test_demo02, test_demo03, test_demo04

# if __name__ == '__main__':
#     # 创建测试套件
#     my_suite = unittest.TestSuite()
#     # 创建加载器
#     loader = unittest.TestLoader()
#     # 使用加载器往测试套件中添加用例
#     my_suite.addTests([loader.loadTestsFromModule(test_demo01),
#                        loader.loadTestsFromModule(test_demo02),
#                        loader.loadTestsFromModule(test_demo03),
#                        loader.loadTestsFromModule(test_demo04)])
#     # 使用test_runner运行测试套件
#     unittest.TextTestRunner(verbosity=2).run(my_suite)

# ---------------------------------------------------------------------------------------------------------
import unittest
import unittestreport

if __name__ == '__main__':
    # 创建测试套件
    my_suite = unittest.TestSuite()
    # 创建加载器
    loader = unittest.TestLoader()
    # 测试套件中添加测试用例
    my_suite.addTest(loader.discover(".\case_1", "test*.py"))
    # 运行测试用例
    my_runner = unittestreport.TestRunner(
        suite=my_suite,
        tester="坚果",
        filename="my_report.html",
        report_dir=".\my_abc",
        title="坚果的测试报告",
        desc="我生成测试报告，想写什么就写什么",
        templates=1
    )
    my_runner.run(count=2, interval=1, thread_count=3)